package com.cg.leetcode;

import org.junit.Test;

import java.util.*;

/**
 * 733.图像渲染
 *
 * @program: LeetCode->LeetCode_733
 * @description: 733.图像渲染
 * @author: cg
 * @create: 2021-09-23 22:07
 **/
public class LeetCode_733 {

    @Test
    public void test733() {
        System.out.println(Arrays.deepToString(floodFill(new int[][]{{1, 1, 1}, {1, 1, 0}, {1, 0, 1}}, 1, 1, 2)));
    }

    /**
     * 有一幅以二维整数数组表示的图画，每一个整数表示该图画的像素值大小，数值在 0 到 65535 之间。
     * 给你一个坐标 (sr, sc) 表示图像渲染开始的像素值（行 ，列）和一个新的颜色值 newColor，让你重新上色这幅图像。
     * 为了完成上色工作，从初始坐标开始，记录初始坐标的上下左右四个方向上像素值与初始坐标相同的相连像素点，接着再记录这四个方向上符合条件的像素点与他们对应四个方向上像素值与初始坐标相同的相连像素点，……，重复该过程。将所有有记录的像素点的颜色值改为新的颜色值。
     * 最后返回经过上色渲染后的图像。
     * <p>
     * 示例 1:
     * 输入:
     * image = [[1,1,1],[1,1,0],[1,0,1]]
     * sr = 1, sc = 1, newColor = 2
     * 输出: [[2,2,2],[2,2,0],[2,0,1]]
     * 解析:
     * 在图像的正中间，(坐标(sr,sc)=(1,1)),
     * 在路径上所有符合条件的像素点的颜色都被更改成2。
     * 注意，右下角的像素没有更改为2，
     * 因为它不是在上下左右四个方向上与初始点相连的像素点。
     * <p>
     * 注意:
     * 1) image 和 image[0] 的长度在范围 [1, 50] 内。
     * 2) 给出的初始点将满足 0 <= sr < image.length 和 0 <= sc < image[0].length。
     * 3) image[i][j] 和 newColor 表示的颜色值在范围 [0, 65535]内。
     *
     * @param image
     * @param sr
     * @param sc
     * @param newColor
     * @return
     */
    public int[][] floodFill(int[][] image, int sr, int sc, int newColor) {
        //深度优先实现
        if (image[sr][sc] != newColor) {
            dfs(image, sr, sc, newColor, image[sr][sc]);
        }
        return image;
    }

    private void dfs(int[][] image, int i, int j, int newColor, int oldColor) {
        //边界判定,image[i][j] != oldColor当前值与开始值相同
        if (i < 0 || i >= image.length || j < 0 || j >= image[0].length || image[i][j] != oldColor) {
            return;
        }
        image[i][j] = newColor;
        //上
        dfs(image, i - 1, j, newColor, oldColor);
        //下
        dfs(image, i + 1, j, newColor, oldColor);
        //左
        dfs(image, i, j - 1, newColor, oldColor);
        //右
        dfs(image, i, j + 1, newColor, oldColor);
    }

    /*int[] dx = {1, 0, 0, -1};
    int[] dy = {0, 1, -1, 0};
    public int[][] floodFill(int[][] image, int sr, int sc, int newColor) {
        //广度优先实现
        int oldColor = image[sr][sc];
        if (oldColor == newColor) {
            return image;
        }
        int n = image.length, m = image[0].length;
        Queue<int[]> queue = new LinkedList<>();
        queue.offer(new int[]{sr, sc});
        image[sr][sc] = newColor;
        while (!queue.isEmpty()) {
            int[] cell = queue.poll();
            int x = cell[0], y = cell[1];
            //上下左右四个方向
            //for (int i = 0; i < 4; i++) {
            //    int mx = x + dx[i], my = y + dy[i];
            //    if (mx >= 0 && mx < n && my >= 0 && my < m && image[mx][my] == oldColor) {
            //        queue.offer(new int[]{mx, my});
            //        image[mx][my] = newColor;
            //    }
            //}
            // 上
            if (x > 0 && image[x - 1][y] == oldColor) {
                image[x - 1][y] = newColor;
                queue.offer(new int[]{x - 1, y});
            }
            // 下
            if (x < n - 1 && image[x + 1][y] == oldColor) {
                image[x + 1][y] = newColor;
                queue.offer(new int[]{x + 1, y});
            }
            // 左
            if (y > 0 && image[x][y - 1] == oldColor) {
                image[x][y - 1] = newColor;
                queue.offer(new int[]{x, y - 1});
            }
            // 右
            if (y < m - 1 && image[x][y + 1] == oldColor) {
                image[x][y + 1] = newColor;
                queue.offer(new int[]{x, y + 1});
            }
        }
        return image;
    }*/

}
